Latent Compaction with Test-Time Training
潜在压缩与测试时训练:可训练 KV Cache 压缩
🧭 LLM 长上下文 KV cache 压缩 · 测试时训练 ★ 深度解读
Tsinghua University
将压缩 KV cache 视为可训练参数,用测试时训练拟合注意力与蒸馏,在 32× 压缩下 RULER 达 96.7%
📌 工作参考价值(4 点)
- 商品详情页、用户长期行为序列、长评论等重复使用的长上下文,可离线预压成可训练 KV cache。模型权重冻结,线上多 query 复用,适合推荐/搜索中大量候选文档场景;但注意论文只考核逻辑 KV-pair 预算,实际物理内存与速度需自行测评。
- 测试时训练不需要下游真实 query:用 480 个通用 prompt 与 teacher 全量 cache 生成 rollouts,就能提供 attention 与蒸馏监督。电商可以模板化商品属性、风格、摘要等通用问题,避免依赖真实用户日志。
- 初始化组合可直接复用:KVZip+ 重建分数选 preserver(占 75% 效果最好)、teacher attention 选 anchor、farthest-first 选 diversifier;ablations 表明 preserver 预算分配对准确率影响最大。
- 若把压缩文档 cache 与后续 query cache 合并,必须按 log-normalizer 加权而不是等权平均,否则准确率几乎归零;attention bias 贡献有限但可保留。